package com.shijimo.server.service;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.shijimo.server.domain.File;
import com.shijimo.server.domain.FileExample;
import com.shijimo.server.dto.FileDto;
import com.shijimo.server.dto.PageDto;
import com.shijimo.server.mapper.FileMapper;
import com.shijimo.server.util.CopyUtil;
import com.shijimo.server.util.UuidUtil;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import javax.annotation.Resource;
import java.util.Date;
import java.util.List;

/**
 * @author Dream_飞翔
 * @email 1072876976@qq.com
 */

@Service
public class FileService {
    @Resource
    private FileMapper fileMapper;

    /**
     * 查询列表方法
     *
     * @param pageDto 分页对象
     */
    public void list(PageDto pageDto) {
        // 开启mybatis分页，pageNum从1开始
        // 生效的位置是开启分页后下方的第一个select语句
        PageHelper.startPage(pageDto.getPage(), pageDto.getSize());

        FileExample fileExample = new FileExample();


        // 进行查询
        List<File> fileList = fileMapper.selectByExample(fileExample);

        PageInfo<File> pageInfo = new PageInfo<>(fileList);
        // 查询到的总条数
        pageDto.setTotal(pageInfo.getTotal());

        List<FileDto> fileDtoList = CopyUtil.copyList(fileList, FileDto.class);
        // 设置最终的查询结果
        pageDto.setList(fileDtoList);
    }

    /**
     * 保存修改结果
     * 1. 如果从前端传来的对象中Id有值的话就是更新操作
     * 2. 如果Id为空则说明是进行新增操作
     *
     * @param fileDto
     */
    public void save(FileDto fileDto) {
        File file = CopyUtil.copy(fileDto, File.class);
        File fileDb = selectByKey(fileDto.getKey());
        // 如果数据库中查询不到key所对应的文件
        if (fileDb == null) {
            this.insert(file);
        } else {
            // 将当前文件的分片索引更新为现在的分片索引
            fileDb.setShardIndex(fileDto.getShardIndex());
            this.update(fileDb);
        }
    }

    /**
     * 添加方法
     */
    private void insert(File file) {
        Date now = new Date();
        file.setCreatedAt(now);
        file.setUpdatedAt(now);

        // 设置Id
        file.setId(UuidUtil.getShortUuid());
        fileMapper.insert(file);
    }

    /**
     * 更新方法
     */
    private void update(File file) {
        file.setUpdatedAt(new Date());

        fileMapper.updateByPrimaryKey(file);
    }

    /**
     * 删除方法
     */
    public void delete(String id) {
        fileMapper.deleteByPrimaryKey(id);
    }

    /**
     * 根据传来的key查找文件
     *
     * @param key
     * @return
     */
    public File selectByKey(String key) {
        FileExample example = new FileExample();
        example.createCriteria().andKeyEqualTo(key);
        List<File> fileList = fileMapper.selectByExample(example);
        if (CollectionUtils.isEmpty(fileList)) {
            return null;
        } else {
            return fileList.get(0);
        }
    }

    public FileDto findByKey(String key) {
        return CopyUtil.copy(selectByKey(key), FileDto.class);
    }
}
